"""
使用 LangServe 部署智能体 Web API
"""

from fastapi import FastAPI
from langserve import add_routes
from main import create_agent

# 创建 FastAPI 应用
app = FastAPI(
    title="智能搜索助手 API",
    version="1.0",
    description="基于通义大模型的智能搜索助手 API"
)

# 创建智能体实例
agent = create_agent(debug=False)

# 添加 LangServe 路由
add_routes(
    app,
    agent.get_chain(),
    config_keys=["configurable"],
    path="/chat"
)

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=8000) 